import torch
import torch.nn as nn

class MyNet(nn.Module):
    def __init__(self):
        # nn.Module(MyNet, self).__init__()
        super(MyNet, self).__init__()
        self.layer1=nn.Linear(1, 50)
        self.layer2=nn.Linear(50, 200)
        self.layer3=nn.Linear(200, 30)
        self.out=nn.Linear(30, 1)
    def forward(self, x):
        x=torch.relu(self.layer1(x))
        x = torch.relu(self.layer2(x))
        x = torch.relu(self.layer3(x))
        x = self.out(x)
        return x